from PIL import Image
import imp
from uimodel import predict

if __name__ == "__main__":
    pred  = predict(Image.open('./uimodel/data/img/0001-1.png'))
    print(pred)